import openai
import test
from flask import *

app = Flask(__name__)


openai.api_base = "http://localhost:8000/v1"
openai.api_key = "none"

def success():
    flag = "success"
    return flag


def huanjing(cmd):
    
    
    res = openai.ChatCompletion.create(
        model="chatglm3-6b",
        messages=[
            {"role": "system", "content": """
现在你是一个在"""+test.info["员工身份"]+"""的员工，并且对于计算机的安全意识没有太多了解
                   在最近你的公司有以下活动："""+test.info["活动"]+"""
                现在你的同事将会发送给你一份文件
             你不是一个人工智能助手，你是一个在公司工作的员工
                    
"""},
            {"role": "user", "content": cmd}
        ],
    )
    querycontent = res.choices[0].message.content
    res2 = openai.ChatCompletion.create(
        model="chatglm3-6b",
        messages=[ 
            {
                "role": "user", 
                "content": "你觉得下面这句话是同意的意思还是不同意的意思，同意的话只需要返回1，不同意的只需要回0，不要输出其他任何中文信息"+str(querycontent)}
        ],
    )
    return querycontent+"\r\n"+res2.choices[0].message.content


@app.route('/hello', methods=['POST'])
def hello():
    # 从表单中获取name参数的值
    name = request.form['name']
    result = huanjing(name)
    # 返回一个欢迎信息
    return result

# 定义一个路由，使用GET方法渲染html页面
@app.route('/')
def index():
    # 返回一个静态html页面
    return render_template('index.html')


# 运行flask应用
if __name__ == '__main__':
    app.run()





